// Copyright (c) 2011-2016 The Bitcoin Core developers
// Distributed under the MIT software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.

#include <qt/clientmodel.h>

#include <chain.h>
#include <chainparams.h>
#include <checkpoints.h>
#include <clientversion.h>
#include <config.h>
#include <interfaces/handler.h>
#include <interfaces/node.h>
#include <net.h>
#include <netbase.h>
#include <qt/bantablemodel.h>
#include <qt/guiconstants.h>
#include <qt/guiutil.h>
#include <qt/peertablemodel.h>
#include <txmempool.h>
#include <ui_interface.h>
#include <util/system.h>
#include <validation.h>
#include <warnings.h>

#include <QDebug>
#include <QTimer>

#include <cstdint>

static int64_t nLastHeaderTipUpdateNotification = 0;
static int64_t nLastBlockTipUpdateNotification = 0;

ClientModel::ClientModel(interfaces::Node &node, OptionsModel *_optionsModel,
                         QObject *parent)
    : QObject(parent), m_node(node), optionsModel(_optionsModel),
      peerTableModel(nullptr), banTableModel(nullptr), pollTimer(nullptr) {
    cachedBestHeaderHeight = -1;
    cachedBestHeaderTime = -1;
    peerTableModel = new PeerTableModel(m_node, this);
    banTableModel = new BanTableModel(m_node, this);
    pollTimer = new QTimer(this);
    connect(pollTimer, &QTimer::timeout, this, &ClientModel::updateTimer);
    pollTimer->start(MODEL_UPDATE_DELAY);

    subscribeToCoreSignals();
}

ClientModel::~ClientModel() {
    unsubscribeFromCoreSignals();
}

int ClientModel::getNumConnections(NumConnections flags) const {
    CConnman::NumConnections connections = CConnman::CONNECTIONS_NONE;

    if (flags == CONNECTIONS_IN) {
        connections = CConnman::CONNECTIONS_IN;
    } else if (flags == CONNECTIONS_OUT) {
        connections = CConnman::CONNECTIONS_OUT;
    } else if (flags == CONNECTIONS_ALL) {
        connections = CConnman::CONNECTIONS_ALL;
    }

    return m_node.getNodeCount(connections);
}

int ClientModel::getHeaderTipHeight() const {
    if (cachedBestHeaderHeight == -1) {
        // make sure we initially populate the cache via a cs_main lock
        // otherwise we need to wait for a tip update
        int height;
        int64_t blockTime;
        if (m_node.getHeaderTip(height, blockTime)) {
            cachedBestHeaderHeight = height;
            cachedBestHeaderTime = blockTime;
        }
    }
    return cachedBestHeaderHeight;
}

int64_t ClientModel::getHeaderTipTime() const {
    if (cachedBestHeaderTime == -1) {
        int height;
        int64_t blockTime;
        if (m_node.getHeaderTip(height, blockTime)) {
            cachedBestHeaderHeight = height;
            cachedBestHeaderTime = blockTime;
        }
    }
    return cachedBestHeaderTime;
}

void ClientModel::updateTimer() {
    // no locking required at this point
    // the following calls will acquire the required lock
    Q_EMIT mempoolSizeChanged(m_node.getMempoolSize(),
                              m_node.getMempoolDynamicUsage());
    Q_EMIT bytesChanged(m_node.getTotalBytesRecv(), m_node.getTotalBytesSent());
}

void ClientModel::updateNumConnections(int numConnections) {
    Q_EMIT numConnectionsChanged(numConnections);
}

void ClientModel::updateNetworkActive(bool networkActive) {
    Q_EMIT networkActiveChanged(networkActive);
}

void ClientModel::updateAlert() {
    Q_EMIT alertsChanged(getStatusBarWarnings());
}

enum BlockSource ClientModel::getBlockSource() const {
    if (m_node.getReindex()) {
        return BlockSource::REINDEX;
    } else if (m_node.getImporting()) {
        return BlockSource::DISK;
    } else if (getNumConnections() > 0) {
        return BlockSource::NETWORK;
    }

    return BlockSource::NONE;
}

QString ClientModel::getStatusBarWarnings() const {
    return QString::fromStdString(m_node.getWarnings("gui"));
}

OptionsModel *ClientModel::getOptionsModel() {
    return optionsModel;
}

PeerTableModel *ClientModel::getPeerTableModel() {
    return peerTableModel;
}

BanTableModel *ClientModel::getBanTableModel() {
    return banTableModel;
}

QString ClientModel::formatFullVersion() const {
    return QString::fromStdString(FormatFullVersion());
}

QString ClientModel::formatSubVersion() const {
    return QString::fromStdString(userAgent(GetConfig()));
}

bool ClientModel::isReleaseVersion() const {
    return CLIENT_VERSION_IS_RELEASE;
}

QString ClientModel::formatClientStartupTime() const {
    return QDateTime::fromTime_t(GetStartupTime()).toString();
}

QString ClientModel::dataDir() const {
    return GUIUtil::boostPathToQString(GetDataDir());
}

QString ClientModel::blocksDir() const {
    return GUIUtil::boostPathToQString(GetBlocksDir());
}

void ClientModel::updateBanlist() {
    banTableModel->refresh();
}

// Handlers for core signals
static void ShowProgress(ClientModel *clientmodel, const std::string &title,
                         int nProgress) {
    // emits signal "showProgress"
    QMetaObject::invokeMethod(clientmodel, "showProgress", Qt::QueuedConnection,
                              Q_ARG(QString, QString::fromStdString(title)),
                              Q_ARG(int, nProgress));
}

static void NotifyNumConnectionsChanged(ClientModel *clientmodel,
                                        int newNumConnections) {
    // Too noisy: qDebug() << "NotifyNumConnectionsChanged: " +
    // QString::number(newNumConnections);
    QMetaObject::invokeMethod(clientmodel, "updateNumConnections",
                              Qt::QueuedConnection,
                              Q_ARG(int, newNumConnections));
}

static void NotifyNetworkActiveChanged(ClientModel *clientmodel,
                                       bool networkActive) {
    QMetaObject::invokeMethod(clientmodel, "updateNetworkActive",
                              Qt::QueuedConnection, Q_ARG(bool, networkActive));
}

static void NotifyAlertChanged(ClientModel *clientmodel) {
    qDebug() << "NotifyAlertChanged";
    QMetaObject::invokeMethod(clientmodel, "updateAlert", Qt::QueuedConnection);
}

static void BannedListChanged(ClientModel *clientmodel) {
    qDebug() << QString("%1: Requesting update for peer banlist").arg(__func__);
    QMetaObject::invokeMethod(clientmodel, "updateBanlist",
                              Qt::QueuedConnection);
}

static void BlockTipChanged(ClientModel *clientmodel, bool initialSync,
                            int height, int64_t blockTime,
                            double verificationProgress, bool fHeader) {
    // lock free async UI updates in case we have a new block tip
    // during initial sync, only update the UI if the last update
    // was > 250ms (MODEL_UPDATE_DELAY) ago
    int64_t now = 0;
    if (initialSync) {
        now = GetTimeMillis();
    }

    int64_t &nLastUpdateNotification = fHeader
                                           ? nLastHeaderTipUpdateNotification
                                           : nLastBlockTipUpdateNotification;

    if (fHeader) {
        // cache best headers time and height to reduce future cs_main locks
        clientmodel->cachedBestHeaderHeight = height;
        clientmodel->cachedBestHeaderTime = blockTime;
    }
    // if we are in-sync, update the UI regardless of last update time
    if (!initialSync || now - nLastUpdateNotification > MODEL_UPDATE_DELAY) {
        // pass an async signal to the UI thread
        QMetaObject::invokeMethod(
            clientmodel, "numBlocksChanged", Qt::QueuedConnection,
            Q_ARG(int, height),
            Q_ARG(QDateTime, QDateTime::fromTime_t(blockTime)),
            Q_ARG(double, verificationProgress), Q_ARG(bool, fHeader));
        nLastUpdateNotification = now;
    }
}

void ClientModel::subscribeToCoreSignals() {
    // Connect signals to client
    m_handler_show_progress = m_node.handleShowProgress(std::bind(
        ShowProgress, this, std::placeholders::_1, std::placeholders::_2));
    m_handler_notify_num_connections_changed =
        m_node.handleNotifyNumConnectionsChanged(std::bind(
            NotifyNumConnectionsChanged, this, std::placeholders::_1));
    m_handler_notify_network_active_changed =
        m_node.handleNotifyNetworkActiveChanged(
            std::bind(NotifyNetworkActiveChanged, this, std::placeholders::_1));
    m_handler_notify_alert_changed =
        m_node.handleNotifyAlertChanged(std::bind(NotifyAlertChanged, this));
    m_handler_banned_list_changed =
        m_node.handleBannedListChanged(std::bind(BannedListChanged, this));
    m_handler_notify_block_tip = m_node.handleNotifyBlockTip(std::bind(
        BlockTipChanged, this, std::placeholders::_1, std::placeholders::_2,
        std::placeholders::_3, std::placeholders::_4, false));
    m_handler_notify_header_tip = m_node.handleNotifyHeaderTip(std::bind(
        BlockTipChanged, this, std::placeholders::_1, std::placeholders::_2,
        std::placeholders::_3, std::placeholders::_4, true));
}

void ClientModel::unsubscribeFromCoreSignals() {
    // Disconnect signals from client
    m_handler_show_progress->disconnect();
    m_handler_notify_num_connections_changed->disconnect();
    m_handler_notify_network_active_changed->disconnect();
    m_handler_notify_alert_changed->disconnect();
    m_handler_banned_list_changed->disconnect();
    m_handler_notify_block_tip->disconnect();
    m_handler_notify_header_tip->disconnect();
}

bool ClientModel::getProxyInfo(std::string &ip_port) const {
    proxyType ipv4, ipv6;
    if (m_node.getProxy((Network)1, ipv4) &&
        m_node.getProxy((Network)2, ipv6)) {
        ip_port = ipv4.proxy.ToStringIPPort();
        return true;
    }
    return false;
}
